"""
-*- coding =utf-8 -*-
@Project : 脚本文件python
#@Author : Crabin
#@File : bhnet.py
#@Time : 2022/10/13 17:08

"""

# 实现netcat的部分功能

import sys
import socket
import getopt
import threading
import subprocess

# 全局变量
listen = False
command = False
upload = False
execute = ""
target = ""
upload_destination = ""
port = ""


# 发送数据
def client_sander(buffer):
    client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    try:
        # 连接主机
        client.connect((target, port))

        if len(buffer):
            client.send(buffer)

        while True:

            # 等待数据回传
            recv_len = 1
            response = ""

            while recv_len:
                data = client.recv(4096)
                recv_len = len(data)
                response += data

                if recv_len < 4096:
                    break

            print(response)

            # 等待更多的输入
            buffer = raw_input("")
            buffer += "\n"

            client.send(buffer)

    except:
        print("[*] Exception Exiting")

    client.close()


def server_loop():
    global target
    global port

    # 本地的，没有目标的话就监听所有接口
    if not len(target):
        target = "0:0:0:0"

    server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    server.bind((target, port))
    server.listen(5)

    while True:
        client_socket, addr = server.accept()  # 数据，主机信息

        # 拆分一个线程处理新的客户端
        client_thread = threading.Thread(target=client_handler, args=(client_socket,))
        client_thread.start()


# 命令处理 shell
def rum_command(command):
    # 换行
    command = command.rstrop()

    # 运行命令并返回结果
    try:
        output = subprocess.check_output(command, stderr=subprocess.STDOUT, shell=True)
    except:
        output = "Failed to execute command.\r\n"

    return output


# 文件上传，命令执行，shell
def client_handler(client_socket):
    global upload
    global execute
    global command

    # 检测上传文件
    if len(upload_destination):

        file_buffer = ""

        # 读取
        while True:
            data = client_socket.recv(1024)

            if not data:
                break
            else:
                file_buffer += data

        # 接收并写入文件
        try:
            file_descriptor = open(upload_destination, "wb")
            file_descriptor.write(file_buffer)
            file_descriptor.close()

            # 回传确认信息
            client_socket.send("Successfully saved file to %s \r\n" % upload_destination)
        except:
            client_socket.send("Failed to save file to %s \r\n" % upload_destination)

    # 检查命令执行
    if len(execute):
        # 运行命令
        output = rum_command(execute)

        client_socket.send(output)

    # 如果需要一个命令行shell ，进入另一个循环

    if command:
        while True:

            # 跳出一个窗口
            client_socket.send("<BHP:#>")

            # 接收文件，知道我们发现换行符
            cmd_buffer = ""
            while "\n" not in cmd_buffer:
                cmd_buffer += client_socket.recv(1024)

            # 运行命令
            response = rum_command(cmd_buffer)

            # 回传响应数据
            client_socket.send(response)



def usage():
    print("BHP Net Tool")
    print("Netcat Replacement")
    print()
    print("Usage: bhpnet.py -t target_host -p port")
    print("-l --listen                - listen on [host]:[port] for incoming connections")
    print("-e --execute=file_to_run   - execute the given file upon receiving a connection")
    print("-c --command               - initialize a command shell")
    print("-u --upload=destination    - upon receiving connection upload a file and write to [destination]")
    print()
    print()
    print("Examples: ")
    print("bhpnet.py -t 192.168.0.1 -p 5555 -l -c")
    print("bhpnet.py -t 192.168.0.1 -p 5555 -l -u=c:\\target.exe")
    print("bhpnet.py -t 192.168.0.1 -p 5555 -l -e=\"cat /etc/passwd\"")
    print("echo 'ABCDEFGHI' | ./bhpnet.py -t 192.168.11.12 -p 135")
    sys.exit(0)


def main():
    global listen
    global port
    global execute
    global command
    global upload_destination
    global target

    if not len(sys.argv[1:]):
        usage()

    # 读取命令行选项
    try:
        opts, args = getopt.getopt(sys.argv[1:], "hle:t:p:cu:",
                                   ["help", "listen", "execute", "target", "port", "command", "upload"])
    except getopt.GetoptError as err:
        print(str(err))
        usage()

    for o, a in opts:
        if o in ("-h", "--help"):
            usage()
        elif o in ("-l", "--listen"):
            listen = True
        elif o in ("-e", "execute"):
            execute = a
        elif o in ("-c", "--commandshell"):
            command = True
        elif o in ("-t", "--target"):
            target = a[1:]
        elif o in ("-p", "--port"):
            port = int(a)

    print(target)

    # 我们是进行监听还是仅从标准输入发送数据？
    if not listen and len(target) and port > 0:
        # 标准输入发送数据
        # 读取内存数据从命令行
        # 这里将阻塞，
        buffer = sys.stdin.read()

        # 发送数据
        client_sander(buffer)

    # 监听
    if listen:
        server_loop()


main()
